# 实现两个数的加减乘除
# 定义加法
def add(a, b):
    return a + b


# 定义减法
def minus(a, b):
    return a - b


# 乘法
def mul(a, b):
    return a * b


# 除法
def divide(a, b):
    if b == 0:
        print("除数不能为0")
        return
    return a / b


# 主函数
if __name__ == '__main__':
    code = int(input("""
    1、加法
    2、减法
    3、乘法
    4、除法
请输入对应的操作码："""))
    a = int(input("请输入第一个数字："))
    b = int(input("请输入第二个数字："))
    if code == 1:
        print(add(a, b))
    elif code == 2:
        print(minus(a, b))
    elif code == 3:
        print(mul(a, b))
    elif code == 4:
        print(divide(a, b))
    else:
        print("请输入正确的操作码")
